"""
给定一个二叉树，返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

示例:

输入:

   1
 /   \
2     3
 \
  5

输出: ["1->2->5", "1->3"]

解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
"""

"""
自己犯错：

没看清楚题目：返回条件写错了，是List[str],我返回了List[List[int]]

递归返回的时候没有把当前的val带上
"""
from leetcode_test import TreeNode
from typing import List

class Solution:
    def binaryTreePaths(self, root: TreeNode) -> List[str]:
        ans = []
        if root is None:
            return []
        def helper(root, res):
            print("root=", root)
            if not root:
                return
            if not root.left and not root.right:
                ans.append("->".join(res + [str(root.val)]))
            helper(root.left, res + [str(root.val)])
            helper(root.right, res + [str(root.val)])
        helper(root, [])
        return ans

def test():
    t = TreeNode.create([1,2,3,None,5])
    assert Solution().binaryTreePaths(t) == ['1->2->5', '1->3']
            
if __name__ == "__main__":
    t = TreeNode.create([1,2,3,None,5])
    Solution().binaryTreePaths(t)